# -*- coding:utf-8 -*-
import random
import time
from module_page.base_page import BaseTrainPage
from get_project_path import project_path


class TrainEnlistPage(BaseTrainPage):
    def add_train(self):  # 新增培训报名
        self.driver.click(self.page_eles['新增培训报名'])
        self.driver.click(self.page_eles['报名集团输入框'])
        self.driver.click(self.page_eles['组织单选总部'])
        self.driver.click(self.page_eles['组织单选确定按钮'])
        self.driver.send_keys(self.page_eles['报名名称输入框'], self.getdata.text(20))
        self.driver.file_send_keys(self.page_eles['附件上传'], project_path + '\data\9.jpeg')


class InformationPage(BaseTrainPage):
    def importadd_information(self, phone_number, ID_number):  # 导入excel新增报名信息
        self.click('新增按钮')
        self.send_keys('姓名输入框', self.getdata.name())
        self.click('性别选择框')
        li_choice = ['性别男', '性别女']
        self.click(random.choice(li_choice))
        self.send_keys('所在企业输入框', self.getdata.text(10))
        self.click('企业级次输入框')
        li_choice = ['总部级次', '省级次', '地市级次', '县级级次', '街道级次']
        self.click(random.choice(li_choice))
        self.send_keys('职务输入框', self.getdata.text(10))
        self.send_keys('职业资格输入框', self.getdata.text(10))
        self.click('职称级别框')
        li_choice = ['初级别', '中级别', '高级别', '副高级别', '正高级别']
        self.click(random.choice(li_choice))
        self.send_keys('手机号输入框', phone_number)
        self.click('身份证号码输入框')
        time.sleep(1)
        if self.driver.page_contains('该用户手机号已存在系统账号') == True:
            self.click('提示确认按钮')
        self.send_keys('身份证号码输入框', ID_number)
        self.click('培训班名称输入框')
        self.click('未开始培训班')
        self.click('培训班确定按钮')
        self.click('培训班次输入框')
        li_choice = ['第一期班次', '第二期班次', '第三期班次', '第四期班次']
        self.click(random.choice(li_choice))
        self.click('保存按钮')

    def randomadd_information(self):  # 随机新增报名信息
        self.click('新增按钮')
        self.send_keys('姓名输入框', self.getdata.name())
        self.click('性别选择框')
        li_choice = ['性别男', '性别女']
        self.click(random.choice(li_choice))
        self.send_keys('所在企业输入框', self.getdata.text(10))
        self.click('企业级次输入框')
        li_choice = ['总部级次', '省级次', '地市级次', '县级级次', '街道级次']
        self.click(random.choice(li_choice))
        self.send_keys('职务输入框', self.getdata.text(10))
        self.send_keys('职业资格输入框', self.getdata.text(10))
        self.click('职称级别框')
        li_choice = ['初级别', '中级别', '高级别', '副高级别', '正高级别']
        self.click(random.choice(li_choice))
        self.send_keys('手机号输入框', self.getdata.phone_number())
        self.send_keys('身份证号码输入框', self.getdata.ID_number())
        self.click('培训班名称输入框')
        self.click('未开始培训班')
        self.click('培训班确定按钮')
        self.click('培训班次输入框')
        li_choice = ['第一期班次', '第二期班次', '第三期班次', '第四期班次']
        self.click(random.choice(li_choice))
        self.click('保存按钮')

    def edit_information(self):  # 编辑报名信息
        self.click('第一条报名信息')
        self.click('编辑按钮')
        self.send_keys('姓名输入框', '刘玉婷')
        self.click('性别选择框')
        self.click('性别男')
        self.send_keys('所在企业输入框', '编辑后企业')
        self.click('企业级次输入框')
        self.click('地市级次')
        self.send_keys('职务输入框', '编辑后职务')
        self.send_keys('职业资格输入框', '编辑后职业资格')
        self.click('职称级别框')
        self.click('中级别')
        self.send_keys('手机号输入框', '13557037909')
        self.send_keys('身份证号码输入框', '14062119770101311X')
        self.click('培训班名称输入框')
        self.click('培训班0707')
        self.click('培训班确定按钮')
        self.click('培训班次输入框')
        self.click('第四期班次')
        self.click('保存按钮')

    def del_information(self):
        self.click('第一条报名信息')
        del_before = self.get_text('第一行的姓名')
        self.click('删除按钮')
        self.click('删除信息的取消按钮')
        del_after = self.get_text('第一行的姓名')
        assert self.driver.text_equal(del_after,del_before)
        self.click('删除按钮')
        self.click('删除信息的确定按钮')
        del_after = self.get_text('第一行的姓名')
        assert self.driver.text_unequal(del_after,del_before)
